# -*- coding: utf-8 -*-
"""
part4: 日志系统， 发布-订阅, 选择性订阅子消息
接受
"""
import pika
import sys

connection = pika.BlockingConnection(
    pika.ConnectionParameters(host='172.22.23.252')
)
channel = connection.channel()

channel.exchange_declare(exchange='direct_logs', exchange_type='direct')

result = channel.queue_declare('', exclusive=True)
queue_name = result.method.queue

severities = sys.argv[1:]

if not severities:
    sys.stderr.write("Usage: %s [info] [waring] [error]\n" % sys.argv[0])
    sys.exit(1)

for severity in severities:
    channel.queue_bind(
        exchange='direct_logs', queue=queue_name, routing_key=severity
    )

print('[*] Waiting for logs. To exit press CTRL+C')


def callback(ch, method, properties, body):
    print("[x] %r: %r" % (method.routing_key, body))


channel.basic_consume(queue=queue_name, on_message_callback=callback, auto_ack=True)

channel.start_consuming()
